package com.camel.gateway.config;

import cn.dev33.satoken.reactor.filter.SaReactorFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @author camel
 */
@Configuration
public class SaTokenConfigure {
    /**
     *  注册 Sa-Token全局过滤器
     **/
    @Bean
    public SaReactorFilter getSaReactorFilter() {
        return new SaReactorFilter()
                // 拦截地址
                // 拦截全部path
                .addInclude("/**")
                // 开放地址
                .addExclude("/favicon.ico")
                // 鉴权方法：每次访问进入
                .setAuth(obj -> {
                    System.out.println("token" + StpUtil.getTokenValue());
                    System.out.println("是否登录" + StpUtil.isLogin());
                    // 登录校验 -- 拦截所有路由，并排除/user/doLogin 用于开放登录
                    SaRouter.match("/**").notMatch("/user/doLogin"
                            , "/api/auth/sso/**"
                            , "/api/system/sso/login"
                            , "/api/auth/sa-res/**").check(r -> StpUtil.checkLogin());

                    // 权限认证 -- 不同模块, 校验不同权限
                    SaRouter.match("/api/system/sysParam/**", r -> StpUtil.checkPermission("system/Variable"));
                    SaRouter.match("/api/system/sysUser/**", r -> StpUtil.checkPermission("system/user"));
                    SaRouter.match("/api/system/sysRole/**", r -> StpUtil.checkPermission("system/Role"));
                    SaRouter.match("/api/system/sysMenu/**", r -> StpUtil.checkPermission("system/Module"));
                    SaRouter.match("/api/system/sysCompany/**", r -> StpUtil.checkPermission("system/Dept"));

                    // 更多匹配 ...  */
                })
                // 异常处理方法：每次setAuth函数出现异常时进入
                .setError(e -> {
                    SaResult error = SaResult.error(e.getMessage());
                    error.setCode(401);
                    return error;
                });
    }
}
